Use index of one array to extract value from list to be appended into new array (Python)